HTMLify
Longest Sub-Array with Sum K.java
Views: 1 | Author: cody
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 | // Longest Sub-Array with Sum K java solution import java.io.*; import java.util.*; class Array { // Driver code public static void main (String[] args) throws IOException{ // Taking input using buffered reader BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); int testcases = Integer.parseInt(br.readLine()); // looping through all testcases while(testcases-- > 0){ String line = br.readLine(); String[] element = line.trim().split("\\s+"); int sizeOfArray = Integer.parseInt(element[0]); int K = Integer.parseInt(element[1]); int arr [] = new int[sizeOfArray]; line = br.readLine(); String[] elements = line.trim().split("\\s+"); for(int i = 0;i<sizeOfArray;i++){ arr[i] = Integer.parseInt(elements[i]); } Solution obj = new Solution(); int res = obj.lenOfLongSubarr(arr, sizeOfArray, K); System.out.println(res); } } } //User function Template for Java class Solution { // Function for finding maximum and value pair public static int lenOfLongSubarr(int arr[], int n, int k) { // Complete the function HashMap<Integer, Integer> map = new HashMap<>(); int sum = 0, maxLen = 0; for (int i = 0; i < n; i++) { sum += arr[i]; if (sum == k) maxLen = i + 1; if (!map.containsKey(sum)) { map.put(sum, i); } if (map.containsKey(sum - k)) { if (maxLen < (i - map.get(sum - k))) maxLen = i - map.get(sum - k); } } return maxLen; } } |